// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Author: wsfuyibing <682805@qq.com>
// Date: 2024-12-02

package edit

import (
	"context"
	"gitee.com/go-libs/result"
	"gitee.com/go-libs/validator-iris/validator"
	"gitee.com/go-wares/framework-iris/framework/src/logics"
	"github.com/kataras/iris/v12"
	"gomq/app"
	"gomq/app/errs"
	"gomq/app/models"
	"gomq/app/services"
)

type Logic struct {
	request  *Request
	response *Response
}

func New() logics.Logical {
	return &Logic{
		request:  &Request{},
		response: &Response{},
	}
}

func (o *Logic) Before(_ context.Context, i iris.Context) (err error) {
	return validator.Json(i, o.request)
}

func (o *Logic) Run(ctx context.Context, _ iris.Context) result.Result {
	var (
		bean    *models.Task
		err     error
		req     = &models.Task{}
		res     = result.New()
		service = services.NewTaskService()
	)

	if bean, err = service.GetById(ctx, o.request.Id); err != nil {
		return res.WithError(errs.ErrDatabaseError)
	}

	if bean == nil {
		return res.WithError(errs.ErrTaskNotFound)
	}

	req.Id = o.request.Id

	req.Topic = o.request.Topic
	req.Tag = o.request.Tag
	req.Concurrency = app.AnyInt32(o.request.Concurrency, 1)
	req.DelaySeconds = app.AnyInt(o.request.DelaySeconds, 0)
	req.Threads = app.AnyInt(o.request.Threads, 1)
	req.MaxRetry = app.AnyInt(o.request.MaxRetry, 3)
	req.Store = o.request.Store
	req.Title = o.request.Title
	req.Remark = o.request.Remark
	req.ConditionExpression = o.request.ConditionExpression

	// Main handler.
	req.Handler = o.request.Handler
	req.HandlerMethod = o.request.HandlerMethod
	req.HandlerTimeout = app.AnyInt(o.request.HandlerTimeout, 5)
	req.HandlerFormat = o.request.HandlerFormat
	req.HandlerResponseFormat = o.request.HandlerResponseFormat
	req.HandlerResponseExpression = o.request.HandlerResponseExpression

	// Failed notification.
	req.Failed = o.request.Failed
	req.FailedMethod = o.request.FailedMethod
	req.FailedTimeout = app.AnyInt(o.request.FailedTimeout, 5)
	req.FailedResponseFormat = o.request.FailedResponseFormat
	req.FailedResponseExpression = o.request.FailedResponseExpression

	// Succeed  notification
	req.Succeed = o.request.Succeed
	req.SucceedMethod = o.request.SucceedMethod
	req.SucceedTimeout = app.AnyInt(o.request.SucceedTimeout, 5)
	req.SucceedResponseFormat = o.request.SucceedResponseFormat
	req.SucceedResponseExpression = o.request.SucceedResponseExpression

	// Send process.
	if _, err = services.NewTaskService().EditById(ctx, o.request.Id, req); err != nil {
		res.WithError(errs.ErrDatabaseError)
		return res
	}

	o.response.Id = bean.Id
	o.response.Topic = bean.Topic
	o.response.Tag = bean.Tag
	return result.New().With(o.response)
}
